package com.xhj.ipersistent.parsing;

/**
 * GenericTokenParser
 *
 * @author XJks
 * @description 通用的令牌解析器，用于解析带有特定起止标记的字符串。
 * 例如：将文本中的 SELECT * FROM user WHERE id = #{id} 解析为实际的值，即SELECT * FROM user WHERE id = ?
 * 其中 ？占位符的值由 TokenHandler 处理器确定。
 */
public class GenericTokenParser {

    // 起始标记，如 "#{"
    private final String openToken;
    // 结束标记，如 "}"
    private final String closeToken;
    // 处理被标记内容的处理器
    private final TokenHandler handler;

    /**
     * 构造方法，初始化起始标记、结束标记和处理器
     *
     * @param openToken  起始标记
     * @param closeToken 结束标记
     * @param handler    处理被标记内容的处理器
     */
    public GenericTokenParser(String openToken, String closeToken, TokenHandler handler) {
        this.openToken = openToken;
        this.closeToken = closeToken;
        this.handler = handler;
    }

    /**
     * 解析传入的文本，将被标记的内容交给 handler 处理后替换
     * 支持转义符 \ 来跳过标记
     *
     * @param text 待解析的字符串
     * @return 解析后的字符串
     */
    public String parse(String text) {
        if (text == null || text.isEmpty()) {
            // 如果文本为空，直接返回空字符串
            return "";
        }
        // 查找起始标记
        int start = text.indexOf(openToken);
        if (start == -1) {
            // 没有起始标记，直接返回原文本
            return text;
        }
        char[] src = text.toCharArray();
        int offset = 0; // 当前解析位置
        final StringBuilder builder = new StringBuilder(); // 最终结果
        StringBuilder expression = null; // 存储被标记的内容，即 openToken 和 closeToken 之间的内容
        do {
            // 判断起始标记是否被转义
            if (start > 0 && src[start - 1] == '\\') {
                // 被转义，去掉反斜杠，保留标记
                builder.append(src, offset, start - offset - 1).append(openToken);
                offset = start + openToken.length();
            } else {
                // 找到未转义的起始标记，准备查找结束标记
                if (expression == null) {
                    expression = new StringBuilder();
                } else {
                    expression.setLength(0);
                }
                // 追加起始标记前的内容
                builder.append(src, offset, start - offset);
                offset = start + openToken.length();
                int end = text.indexOf(closeToken, offset);
                // 查找结束标记，处理转义情况
                while (end > -1) {
                    if (end > offset && src[end - 1] == '\\') {
                        // 结束标记被转义，去掉反斜杠，保留标记
                        expression.append(src, offset, end - offset - 1).append(closeToken);
                        offset = end + closeToken.length();
                        end = text.indexOf(closeToken, offset);
                    } else {
                        // 找到未转义的结束标记
                        expression.append(src, offset, end - offset);
                        break;
                    }
                }
                if (end == -1) {
                    // 没有找到结束标记，剩余内容全部追加
                    builder.append(src, start, src.length - start);
                    offset = src.length;
                } else {
                    // 将被标记内容交给 handler 处理，并追加处理结果
                    builder.append(handler.handleToken(expression.toString()));
                    offset = end + closeToken.length();
                }
            }
            // 继续查找下一个起始标记
            start = text.indexOf(openToken, offset);
        } while (start > -1);
        // 追加剩余内容
        if (offset < src.length) {
            builder.append(src, offset, src.length - offset);
        }
        return builder.toString();
    }
}